package leetcode.code0398;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Solution {

	Map<Integer, List<Integer>> map;

	public Solution(int[] nums) {
		map = new HashMap<>();
		for (int i = 0; i < nums.length; i++) {
			int cur = nums[i];
			List<Integer> list = map.getOrDefault(cur, new ArrayList<>());
			list.add(i);
			map.put(cur, list);
		}
	}

	public int pick(int target) {
		List<Integer> list = map.get(target);
		return list.get((int) (Math.random() * list.size()));
	}
	

}
